/*
 * Copyright 2009 Prime Technology.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.primefaces.facestrace.application;

import java.util.Map;

import javax.faces.application.ViewHandler;
import javax.faces.application.ViewHandlerWrapper;
import javax.faces.component.UIViewRoot;
import javax.faces.context.FacesContext;

import org.primefaces.facestrace.component.Trace;

public class FacesTraceViewHandler extends ViewHandlerWrapper{

	private ViewHandler baseViewHandler;
	
	public FacesTraceViewHandler(ViewHandler baseViewHandler) {
		this.baseViewHandler = baseViewHandler;
	}

	protected ViewHandler getWrapped() {
		return this.baseViewHandler;
	}

	@Override
	public UIViewRoot createView(FacesContext facesContext, String viewId) {
		UIViewRoot viewRoot = baseViewHandler.createView(facesContext, viewId);
		Map<String,String> params = facesContext.getExternalContext().getRequestParameterMap();
		String traceParam = params.get("trace");
		
		if(traceParam != null && traceParam.equalsIgnoreCase("true")) {
			Trace trace = (Trace) facesContext.getApplication().createComponent(Trace.COMPONENT_TYPE);
			trace.setRendererType(Trace.RENDERER_TYPE);
			trace.setId("uitrace");
			
			viewRoot.getChildren().add(trace);
		}

		return viewRoot;
	}
}